package com.lyl.myaiagent.optimize.enhancement_association;

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;

/**
 * @author lyl
 * @version 1.0
 * @CreateTime: 2025-08-19
 */
public class ContextualQueryAugmenterFactory {

    public static ContextualQueryAugmenter createInstance(){
        PromptTemplate promptTemplate = new PromptTemplate("""
                你应该输出下面的内容：
                抱歉，我只能回答中国海上交通法的相关问题，别的没办法帮助你哦，
                想要了解详细信息，可以联系客服：150XXXXXXXX;
                """);

        return ContextualQueryAugmenter.builder()
                .allowEmptyContext(false)
                .emptyContextPromptTemplate(promptTemplate)
                .build();
    }
}
